# coding = utf-8

import torch


def test():
    cuda = torch.cuda
    cuda.is_available()
    if torch.cuda.is_available():
        print("use cuda")
    else:
        print("no cuda")


if __name__ == '__main__':
    test()